package com.canvas.netty;

import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.handler.stream.ChunkedWriteHandler;

/**
 * @Author: 宗恣
 * @Date: 2018/10/16 0016 9:22
 */
public class WSServerInitialzer extends ChannelInitializer<SocketChannel> {

    @Override
    protected void initChannel(SocketChannel ch) throws Exception {
        ChannelPipeline pipeline = ch.pipeline();

        // websocket 基于http协议，所以要有http编解码
        pipeline.addLast("HttpServerCodec", new HttpServerCodec())
                // 对写大数据流的支持
                .addLast("ChunkedWriteHandler", new ChunkedWriteHandler())
                // 对httpMessage 进行聚合，聚合成FullHttpRequest 或 FullHttpResponse
                // 几乎在netty 中的编码，都会使用到此handler
                .addLast("HttpObjectAggregator", new HttpObjectAggregator(1024 * 64));

        // ==============  以上是用于支持http协议 ==================

        /**
         * websocket 服务器处理的协议， 用于指定给客户端链接访问的路由: /ws
         * 本 handler 会帮你处理一些繁重的复杂的事
         * 会帮你处理握手操作： 握手 handshaking（close, ping, pong） ping + pong = 心跳
         * 对于websocket来讲，都是以frames进行传输的，不同的数据类型对应的frames也不同
         */
        pipeline.addLast("WebSocketServerProtocolHandler", new WebSocketServerProtocolHandler("/ws"));

        // 自定义的handler
        pipeline.addLast(new ChatHandler());

    }
}
